from grafo import *
def main():
	vertices=Vertices()
	print ' En este trabajo practico proyectaremos las distinas '
	print ' propagaciones de una epidemia'
	pos = 0
	for i in vertices.getListaDeVertices():
		print '- ' + str(pos) + ')' + i
		pos = pos + 1 
	ciudad = raw_input (' Elige elige el numero de una ciudad o t para todas ')
	graf = input ( ' elige 1) para peor escenario o 2) para peor camino ' )
	grafo = Grafo()
	if graf == 1:
		if ciudad == 't':
			grafo.bfs_completo()
		else:
			grafo.bfs(int(ciudad))
	elif graf == 2:
		if ciudad == 't':
			grafo.peor_camino_completo()
		else:
			grafo.peor_camino(int(ciudad))
if __name__ =="__main__":
	main()
